/*
 * Copyright 2013-2018 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.szamc.fintech.streamapi.demos.web;

import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.mvc.method.annotation.StreamingResponseBody;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;

/**
 * @author <a href="mailto:chenxilzx1@gmail.com">theonefx</a>
 */
@Controller
public class BasicController {

    // http://127.0.0.1:8080/hello?name=lisi
    @RequestMapping("/hello")
    @ResponseBody
    @CrossOrigin
    public String hello(@RequestParam(name = "name", defaultValue = "unknown user") String name) {
        return "Hello " + name;
    }

    // http://127.0.0.1:8080/user
    @RequestMapping("/user")
    @ResponseBody
    public User user() {
        User user = new User();
        user.setName("theonefx");
        user.setAge(666);
        return user;
    }

    // http://127.0.0.1:8080/save_user?name=newName&age=11
    @RequestMapping("/save_user")
    @ResponseBody
    public String saveUser(User u) {
        return "user will save: name=" + u.getName() + ", age=" + u.getAge();
    }

    // http://127.0.0.1:8080/html
    @RequestMapping("/html")
    public String html() {
        return "index.html";
    }

    @ModelAttribute
    public void parseUser(@RequestParam(name = "name", defaultValue = "unknown user") String name
            , @RequestParam(name = "age", defaultValue = "12") Integer age, User user) {
        user.setName("zhangsan");
        user.setAge(18);
    }



    @CrossOrigin
    @GetMapping("/getAITalk")
    public ResponseEntity<StreamingResponseBody> getAITalk() {
       String content = "网购真是个让人又爱又恨的东西。每次在网上买东西的时候，我都感觉自己像是在寻宝。看着那些商品图片，就像看到了一个个闪闪发光的宝藏。下单的时候，那叫一个兴奋，感觉自己马上就要拥有世界上最棒的东西了。但是，等收到货的时候，就像是打开了一个神秘的盲盒，有时候是惊喜，更多的时候是惊吓。";
        // 模拟数据块
        ArrayList<String> dataList = new ArrayList<>();
        // 创建一个发送句柄
        StreamingResponseBody responseBody = outputStream -> {
            // 模拟真实场景下连续发送数据
            for (int i = 0; i < content.length(); i++) {
                // 延迟等待：模拟真实场景
                try {
                    Thread.sleep(150);
                }catch (Exception e){}
                // 数据块编码
                String data = content.substring(i, i + 1);
                // 写入数据
                outputStream.write(data.getBytes());
                // 发送数据
                outputStream.flush();
            }
        };
        // 响应到客户端
        return ResponseEntity.ok()
                .contentType(MediaType.APPLICATION_OCTET_STREAM)
                .body(responseBody);
    }


}
